package main

import (
	"database/sql"
	"log"
	"net/http"
	"crypto/tls"
	"github.com/go-sql-driver/mysql"
)

// 起動するポート
const port = "8080"

func helloHandler(w http.ResponseWriter, r *http.Request) {
	// データベース接続
	// TiDB Cloudにて表示されたコード
	mysql.RegisterTLSConfig("tidb", &tls.Config{
		MinVersion: tls.VersionTLS12,
		ServerName: "gateway01.ap-northeast-1.prod.aws.tidbcloud.com",
	})
	db, err := sql.Open("mysql", "xxxxxxxxxx.root:xxxxxxxxxx@tcp(gateway01.ap-northeast-1.prod.aws.tidbcloud.com:4000)/test?tls=tidb")
    // エラー処理
	if err != nil {
		http.Error(w, "Failed to connect to database", http.StatusInternalServerError)
		log.Printf("Failed to connect to database: %v", err)
		return
	}
	// 関数終了時にコネクションを閉じる
	defer db.Close()

	// クエリ実行
	rows, err := db.Query("SHOW DATABASES")
	if err != nil {
		http.Error(w, "Failed to execute query", http.StatusInternalServerError)
		log.Printf("Failed to execute query: %v", err)
		return
	}
    // 終了後にクエリ結果をリリース
	defer rows.Close()

	// クエリ結果をログ出力
	for rows.Next() {
		var dbName string
		if err := rows.Scan(&dbName); err != nil {
			log.Printf("Failed to scan row: %v", err)
			return
		}
		log.Printf("%s", dbName)
	}

	if err = rows.Err(); err != nil {
		log.Printf("Error occurred during rows iteration: %v", err)
		return
	}
}

func main() {
	// ハンドラ関数を設定
	http.HandleFunc("/locust-test", helloHandler)
	// ログ出力
	log.Printf("Server is running on port %s\n", port)
	// サーバの起動
	if err := http.ListenAndServe(":"+port, nil); err != nil {
		log.Fatalf("Failed to start server: %v", err)
	}
}

from locust import HttpUser, task, between

class QuickstartUser(HttpUser):
    # ユーザーが次のタスクを実行するまでの待ち時間を1秒から2.5秒の間でランダムに設定
    wait_time = between(1, 2.5)

    # 負荷テスト中に行われるタスク
    @task
    def index_page(self):
        self.client.get("/locust-test")